<template>
  <div class="app-container">
    <!-- 搜索表单 -->
    <el-form :model="queryParams" ref="queryForm" size="small" :inline="true" v-show="showSearch" label-width="80px">
      <el-form-item label="实验ID" prop="experimentId">
        <el-input v-model="queryParams.experimentId" placeholder="请输入实验ID" clearable @keyup.enter.native="handleQuery" />
      </el-form-item>
      <el-form-item label="模型名称" prop="modelName">
        <el-input v-model="queryParams.modelName" placeholder="请输入模型名称" clearable @keyup.enter.native="handleQuery" />
      </el-form-item>
      <el-form-item label="训练状态" prop="status">
        <el-select v-model="queryParams.status" placeholder="请选择训练状态" clearable>
          <el-option label="未开始" value="pending" />
          <el-option label="训练中" value="training" />
          <el-option label="已完成" value="completed" />
          <el-option label="已失败" value="failed" />
        </el-select>
      </el-form-item>
      <el-form-item label="审核状态" prop="reviewStatus">
        <el-select v-model="queryParams.reviewStatus" placeholder="请选择审核状态" clearable>
          <el-option label="待审核" value="pending" />
          <el-option label="已通过" value="approved" />
          <el-option label="已拒绝" value="rejected" />
        </el-select>
      </el-form-item>
      <el-form-item label="训练时间" prop="dateRange">
        <el-date-picker
          v-model="dateRange"
          type="daterange"
          range-separator="至"
          start-placeholder="开始日期"
          end-placeholder="结束日期"
          value-format="yyyy-MM-dd"
          @change="handleDateRangeChange"
        />
      </el-form-item>
      <el-form-item>
        <el-button type="primary" icon="el-icon-search" size="mini" @click="handleQuery">搜索</el-button>
        <el-button icon="el-icon-refresh" size="mini" @click="resetQuery">重置</el-button>
      </el-form-item>
    </el-form>

    <!-- 操作按钮 -->
    <el-row :gutter="10" class="mb8">
      <el-col :span="1.5">
        <el-button type="primary" plain icon="el-icon-plus" size="mini" @click="handleAdd"
          v-hasPermi="['aiexperiments:training:add']">新增</el-button>
      </el-col>
      <el-col :span="1.5">
        <el-button type="success" plain icon="el-icon-edit" size="mini" :disabled="single" @click="handleUpdate"
          v-hasPermi="['aiexperiments:training:edit']">修改</el-button>
      </el-col>
      <el-col :span="1.5">
        <el-button type="danger" plain icon="el-icon-delete" size="mini" :disabled="multiple" @click="handleDelete"
          v-hasPermi="['aiexperiments:training:remove']">删除</el-button>
      </el-col>
      <right-toolbar :showSearch.sync="showSearch" @queryTable="getList"></right-toolbar>
    </el-row>

    <!-- 训练列表 -->
    <el-table v-loading="loading" :data="trainingList" @selection-change="handleSelectionChange">
      <el-table-column type="selection" width="55" align="center" />
      <el-table-column label="训练ID" align="center" prop="trainingId" width="100" />
      <el-table-column label="实验ID" align="center" prop="experimentId" width="100" />
      <el-table-column label="模型名称" align="center" prop="modelName" :show-overflow-tooltip="true" />
      <el-table-column label="训练状态" align="center" prop="status">
        <template slot-scope="scope">
          <el-tag :type="scope.row.status === 'completed' ? 'success' :
            scope.row.status === 'training' ? 'warning' :
            scope.row.status === 'failed' ? 'danger' : 'info'">
            {{ scope.row.status === 'pending' ? '未开始' :
               scope.row.status === 'training' ? '训练中' :
               scope.row.status === 'completed' ? '已完成' : '已失败' }}
          </el-tag>
        </template>
      </el-table-column>
      <el-table-column label="训练开始时间" align="center" prop="trainingStartTime" width="100">
        <template slot-scope="scope">
          <span>{{ parseTime(scope.row.trainingStartTime, '{y}-{m}-{d}') }}</span>
        </template>
      </el-table-column>
      <el-table-column label="训练结束时间" align="center" prop="trainingEndTime" width="100">
        <template slot-scope="scope">
          <span>{{ parseTime(scope.row.trainingEndTime, '{y}-{m}-{d}') }}</span>
        </template>
      </el-table-column>
      <el-table-column label="训练结果" align="center" prop="trainingResult" :show-overflow-tooltip="true" />
      <el-table-column label="审核状态" align="center" prop="reviewStatus">
        <template slot-scope="scope">
          <el-tag :type="scope.row.reviewStatus === 'approved' ? 'success' :
            scope.row.reviewStatus === 'rejected' ? 'danger' : 'info'">
            {{ scope.row.reviewStatus === 'pending' ? '待审核' :
               scope.row.reviewStatus === 'approved' ? '已通过' : '已拒绝' }}
          </el-tag>
        </template>
      </el-table-column>
      <el-table-column label="审核意见" align="center" prop="comments" :show-overflow-tooltip="true" />
      <el-table-column label="操作" align="center" class-name="small-padding fixed-width">
        <template slot-scope="scope">
          <el-button size="mini" type="text" icon="el-icon-edit" @click="handleUpdate(scope.row)"
            v-hasPermi="['aiexperiments:training:edit']">修改</el-button>
          <el-button size="mini" type="text" icon="el-icon-delete" @click="handleDelete(scope.row)"
            v-hasPermi="['aiexperiments:training:remove']">删除</el-button>
        </template>
      </el-table-column>
    </el-table>

    <pagination v-show="total>0" :total="total" :page.sync="queryParams.pageNum" :limit.sync="queryParams.pageSize"
      @pagination="getList" />

    <!-- 添加或修改训练对话框 -->
    <el-dialog :title="title" :visible.sync="open" width="500px" append-to-body>
      <el-form ref="form" :model="form" :rules="rules" label-width="80px">
        <el-form-item label="实验ID" prop="experimentId">
          <el-input v-model="form.experimentId" placeholder="请输入实验ID" />
        </el-form-item>
        <el-form-item label="模型名称" prop="modelName">
          <el-input v-model="form.modelName" placeholder="请输入模型名称" />
        </el-form-item>
        <el-form-item label="算法ID" prop="algorithmId">
          <el-input v-model="form.algorithmId" placeholder="请输入算法ID" />
        </el-form-item>
        <el-form-item label="训练集ID" prop="trainingDatasetId">
          <el-input v-model="form.trainingDatasetId" placeholder="请输入训练集ID" />
        </el-form-item>
        <el-form-item label="验证集ID" prop="validationDatasetId">
          <el-input v-model="form.validationDatasetId" placeholder="请输入验证集ID" />
        </el-form-item>
        <el-form-item label="训练状态" prop="status">
          <el-select v-model="form.status" placeholder="请选择训练状态">
            <el-option label="未开始" value="pending" />
            <el-option label="训练中" value="training" />
            <el-option label="已完成" value="completed" />
            <el-option label="已失败" value="failed" />
          </el-select>
        </el-form-item>
        <el-form-item label="训练结果" prop="trainingResult">
          <el-input v-model="form.trainingResult" type="textarea" placeholder="请输入训练结果" />
        </el-form-item>
        <el-form-item label="审核状态" prop="reviewStatus">
          <el-select v-model="form.reviewStatus" placeholder="请选择审核状态">
            <el-option label="待审核" value="pending" />
            <el-option label="已通过" value="approved" />
            <el-option label="已拒绝" value="rejected" />
          </el-select>
        </el-form-item>
        <el-form-item label="审核意见" prop="comments">
          <el-input v-model="form.comments" type="textarea" placeholder="请输入审核意见" />
        </el-form-item>
      </el-form>
      <div slot="footer" class="dialog-footer">
        <el-button type="primary" @click="submitForm">确 定</el-button>
        <el-button @click="cancel">取 消</el-button>
      </div>
    </el-dialog>
  </div>
</template>

<script>
import { listTraining, getTraining, delTraining, addTraining, updateTraining } from "@/api/aiexperiments/training";

export default {
  name: "Training",
  data() {
    return {
      // 遮罩层
      loading: true,
      // 选中数组
      ids: [],
      // 非单个禁用
      single: true,
      // 非多个禁用
      multiple: true,
      // 显示搜索条件
      showSearch: true,
      // 总条数
      total: 0,
      // 实验训练记录表格数据
      trainingList: [],
      // 弹出层标题
      title: "",
      // 是否显示弹出层
      open: false,
      // 查询参数
      queryParams: {
        pageNum: 1,
        pageSize: 10,
        experimentId: null,
        modelName: null,
        algorithmId: null,
        trainingDatasetId: null,
        validationDatasetId: null,
        status: null,
        trainingStartTime: null,
        trainingEndTime: null,
        trainingResult: null,
        reviewStatus: null,
        reviewDate: null,
        reviewerId: null,
        comments: null,
        createdBy: null,
        createdAt: null,
        updatedBy: null,
        updatedAt: null,
        deletedBy: null,
        deletedAt: null,
        deleteFlag: null,
        version: null
      },
      // 表单参数
      form: {},
      // 表单校验
      rules: {
        experimentId: [
          { required: true, message: "实验ID不能为空", trigger: "blur" }
        ],
        modelName: [
          { required: true, message: "模型名称不能为空", trigger: "blur" }
        ],
        algorithmId: [
          { required: true, message: "算法ID不能为空", trigger: "blur" }
        ],
      }
    };
  },
  created() {
    this.getList();
  },
  methods: {
    /** 查询实验训练记录列表 */
    getList() {
      this.loading = true;
      listTraining(this.queryParams).then(response => {
        this.trainingList = response.rows;
        this.total = response.total;
        this.loading = false;
      });
    },
    // 取消按钮
    cancel() {
      this.open = false;
      this.reset();
    },
    // 表单重置
    reset() {
      this.form = {
        trainingId: null,
        experimentId: null,
        modelName: null,
        algorithmId: null,
        trainingDatasetId: null,
        validationDatasetId: null,
        status: null,
        trainingStartTime: null,
        trainingEndTime: null,
        trainingResult: null,
        reviewStatus: null,
        reviewDate: null,
        reviewerId: null,
        comments: null,
        createdBy: null,
        createdAt: null,
        updatedBy: null,
        updatedAt: null,
        deletedBy: null,
        deletedAt: null,
        deleteFlag: null,
        version: null
      };
      this.resetForm("form");
    },
    /** 搜索按钮操作 */
    handleQuery() {
      this.queryParams.pageNum = 1;
      this.getList();
    },
    /** 重置按钮操作 */
    resetQuery() {
      this.resetForm("queryForm");
      this.handleQuery();
    },
    // 多选框选中数据
    handleSelectionChange(selection) {
      this.ids = selection.map(item => item.trainingId)
      this.single = selection.length!==1
      this.multiple = !selection.length
    },
    /** 新增按钮操作 */
    handleAdd() {
      this.reset();
      this.open = true;
      this.title = "添加实验训练记录";
    },
    /** 修改按钮操作 */
    handleUpdate(row) {
      this.reset();
      const trainingId = row.trainingId || this.ids
      getTraining(trainingId).then(response => {
        this.form = response.data;
        this.open = true;
        this.title = "修改实验训练记录";
      });
    },
    /** 提交按钮 */
    submitForm() {
      this.$refs["form"].validate(valid => {
        if (valid) {
          if (this.form.trainingId != null) {
            updateTraining(this.form).then(response => {
              this.$modal.msgSuccess("修改成功");
              this.open = false;
              this.getList();
            });
          } else {
            addTraining(this.form).then(response => {
              this.$modal.msgSuccess("新增成功");
              this.open = false;
              this.getList();
            });
          }
        }
      });
    },
    /** 删除按钮操作 */
    handleDelete(row) {
      const trainingIds = row.trainingId || this.ids;
      this.$modal.confirm('是否确认删除实验训练记录编号为"' + trainingIds + '"的数据项？').then(function() {
        return delTraining(trainingIds);
      }).then(() => {
        this.getList();
        this.$modal.msgSuccess("删除成功");
      }).catch(() => {});
    },
    /** 导出按钮操作 */
    handleExport() {
      this.download('aiexperiments/training/export', {
        ...this.queryParams
      }, `training_${new Date().getTime()}.xlsx`)
    }
  }
};
</script>
